package com.atguigu.mall.mallorder.config;

import feign.RequestInterceptor;
import feign.RequestTemplate;
import org.springframework.context.annotation.Bean;
import org.springframework.context.annotation.Configuration;
import org.springframework.web.context.request.RequestContextHolder;
import org.springframework.web.context.request.ServletRequestAttributes;

import javax.servlet.http.HttpServletRequest;

/** feign拦截器功能 */
@Configuration
public class FeignConfig {

  @Bean("requestInterceptor")
  public RequestInterceptor requestInterceptor() {

    RequestInterceptor requestInterceptor =
        new RequestInterceptor() {
          @Override
          public void apply(RequestTemplate template) {
            // 1、使用RequestContextHolder拿到刚进来的请求数据
            ServletRequestAttributes requestAttributes =
                (ServletRequestAttributes) RequestContextHolder.getRequestAttributes();

            if (requestAttributes != null) {
              // 老请求
              HttpServletRequest request = requestAttributes.getRequest();

              if (request != null) {
                // 2、同步请求头的数据（主要是cookie）
                // 把老请求的cookie值放到新请求上来，进行一个同步
                String cookie = request.getHeader("Cookie");
                template.header("Cookie", cookie);
              }
            }
          }
        };

    return requestInterceptor;
  }
}
